package nio.niotest;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;

/**
 * @author charwayH
 * 目标:NIO非阻塞通信的入门案例，服务端开发
 */
public class Server {
    public static void main(String[] args) throws IOException {
        System.out.println("======服务端启动=======");
        // 1.获取服务端通道
        ServerSocketChannel ssChannel = ServerSocketChannel.open();
        // 2.切换为非阻塞模式
        ssChannel.configureBlocking(false);
        // 3.绑定连接的端口
        ssChannel.bind(new InetSocketAddress(9999));
        // 4.获取选择器
        Selector selector = Selector.open();
        // 5.将通道都注册到选择器上，并且开始指定监听接收事件
        ssChannel.register(selector, SelectionKey.OP_ACCEPT);
        // 6.使用Selector选择器轮询已经就绪好的事件
        while (selector.select() > 0){
            System.out.println("开始一轮事件处理...");
            // 7.获取选择器中的所有注册通道已经就绪好的事件
            Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
            // 8.开始遍历这些准备好的事件
            while (iterator.hasNext()){
                SelectionKey sk = iterator.next();
                // 9.判断这个事件具体是什么
                if(sk.isAcceptable()){
                    // 10.直接获取当前接入的客户端通道
                    SocketChannel sChannel = ssChannel.accept();
                    // 11.切换非阻塞模式
                    sChannel.configureBlocking(false);
                    // 12.将本客户端通道注册到选择器
                    sChannel.register(selector, SelectionKey.OP_READ);
                }else if(sk.isReadable()){
                    // 13.获取当前选择器上的读就绪事件
                    SocketChannel sChannel = (SocketChannel) sk.channel();
                    // 14.开始读取数据
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    int len =0;
                    while ((len = sChannel.read(buffer)) > 0){
                        buffer.flip();
                        System.out.println(new String(buffer.array(), 0, len));
                        // 清除缓冲区中的内容
                        buffer.clear();
                    }
                }
                // 处理完毕后需要移除当前事件
                iterator.remove();
            }
        }
    }
}
